package study.数据结构.非线性结构.哈希表;

import java.util.Scanner;

public class HashTableDemo {

    public static void main(String[] args) {

        //创建哈希表
        HashTable hashTable = new HashTable(7);
        //写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add:添加雇员");
            System.out.println("list:显示雇员");
            System.out.println("find:查找雇员");
            System.out.println("exit:退出程序");

            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    //创建雇员
                    Emp emp = new Emp(id, name);
                    hashTable.add(emp);
                    break;
                case "list":
                    hashTable.list();
                    break;
                case "find":
                    System.out.println("请输入要查找的id");
                    Scanner s2 = new Scanner(System.in);
                    int i = s2.nextInt();
                    hashTable.findEmpById(i);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                    break;
                default:
                    break;
            }
        }
    }
}

class Emp {  //表示一个雇员
    private Integer id;
    private String name;
    private Emp next;  //next 默认为空

    public Emp(Integer id, String name) {
        this.id = id;
        this.name = name;
    }

    public Integer getId() {
        return id;
    }

    public void setId(Integer id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public Emp getNext() {
        return next;
    }

    public void setNext(Emp next) {
        this.next = next;
    }

    @Override
    public String toString() {
        return "Emp{" +
                "id=" + id +
                ", name='" + name + '\''
                + '}';
    }
}

//创建EmpLinkedList,表示链表
class EmpLinkedList {
    //头指针，指向第一个Emp,因此链表的head是直接指向第一个Emp
    private Emp head;  //默认为空


    //添加员工到链表
    //说明：
    //  1、假定，当添加雇员时，id时自增的，即id的分配总是从小到大。将雇员加到链表的最后
    public void add(Emp emp) {
        //如果时添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        //如果不是第一个雇员，则遍历链表将雇员添加到链表的最后
        Emp curEmp = head; //辅助指针，用来遍历
        while (true) {
            if (curEmp.getNext() == null) {
                break;
            } else {
                curEmp = curEmp.getNext();
            }
        }
        //退出时就是链表最后，添加雇员
        curEmp.setNext(emp);
    }

    //遍历链表的雇员信息
    public void list(int no) {
        if (head == null) {   //说明链表为空
            System.out.println("第" + (no + 1) + "条链表为空");
            return;
        }
        System.out.println("第" + (no + 1) + "条链表为空为:");
        Emp curEmp = head;  //辅助指针用来遍历
        while (true) {
            System.out.printf("=> id=%d name=%s\t", curEmp.getId(), curEmp.getName());
            if (null == curEmp.getNext()) {
                break;
            } else {
                curEmp = curEmp.getNext();
            }
        }
        System.out.println();
    }

    //根据id查找雇员
    //如果查找到就返回Emp,如果没找到就返回null
    public Emp findEmpById(int id) {
        //判断链表是否为空
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        //辅助指针
        Emp curEmp = head;
        while (true) {
            if (curEmp.getId() == id) {//找到
                break;
            }else if (curEmp.getNext()==null){
                //说明遍历当前链表没有找到  Emp
                curEmp = null;
                break;
            }
            curEmp = curEmp.getNext();  //后移
        }
        return curEmp;
    }
}

//创建HashTable  管理多条链表
class HashTable {
    private EmpLinkedList[] empLinkedListArray;
    private int size;// 表示有多少条链表

    public HashTable(int size) {
        this.size = size;
        //初始化empLinkedListArray
        this.empLinkedListArray = new EmpLinkedList[size];
        //不要忘了初始化哈希表中的链表
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i] = new EmpLinkedList();
        }
    }

    //添加雇员
    public void add(Emp emp) {
        //根据员工的id得到该员工应该加入到哪个链表
        int empLinkedListNo = hashFun(emp.getId());
        //将emp添加到对应的链表中
        empLinkedListArray[empLinkedListNo].add(emp);
    }

    //遍历哈希表
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i);
        }
    }

    //根据输入的id查找雇员
    public void findEmpById(int id){
        //使用散列函数  确定到哪条链表查找
        int empLinkedListNo = hashFun(id);
        Emp emp = empLinkedListArray[empLinkedListNo].findEmpById(id);
        if (emp!=null){  //找到
            System.out.printf("在第%d条链表中找到该雇员 id = %d\n",empLinkedListNo+1,id);
        }else {
            System.out.println("在哈希表中没有找到该雇员");
        }
    }

    //散列函数,使用简单的取模实现
    public int hashFun(int id) {
        return id % size;
    }
}



















